package com.leetcode.sorts.arrays.slide_window;

import java.util.HashMap;

/**
 * @Author: WuPeng
 * @Date: 2021/5/21 3:51 下午
 *
 * 904: 水果成篮
 * 在一排树中，第 i 棵树产生tree[i] 型的水果。
 *
 * 你可以从你选择的任何树开始，然后重复执行以下步骤：
 *      把这棵树上的水果放进你的篮子里。如果你做不到，就停下来。
 *      移动到当前树右侧的下一棵树。如果右边没有树，就停下来。
 *      请注意，在选择一颗树后，你没有任何选择：你必须执行步骤 1，然后执行步骤 2，然后返回步骤 1，然后执行步骤 2，依此类推，直至停止。
 *
 * 你有两个篮子，每个篮子可以携带任何数量的水果，但你希望每个篮子只携带一种类型的水果。
 *
 * 用这个程序你能收集的水果树的最大总量是多少？
 *
 */
public class FruitsCollects_904 {

    public static int solution(int[] nums){
        int res = 0, slow = 0;
        Counter counter = new Counter();
        for (int fast = 0; fast < nums.length; fast++) {
            counter.add(nums[fast], 1);
            while (counter.size() >= 3){
                counter.add(nums[slow], -1);
                if (counter.get(nums[slow]) == 0)
                    counter.remove(nums[slow]);
                slow++;
            }
            res = (fast-slow+1) > res ? (fast-slow+1) : res;
        }
        return res;
    }

    static class Counter extends HashMap<Integer, Integer>{
        public int get(int key){
            return containsKey(key) ? super.get(key) : 0;
        }
        public void add(int key, int val){
            put(key, get(key) + val);
        }
    }

    public static void main(String[] args) {
        int[] nums = {0,0,1,2,2,2};
        System.out.println(solution(nums));
    }
}
